package com.dubito.muxin.netty;

import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpClientCodec;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;


/**
 * @author liliang
 * @description:
 * @date 2018/11/15.
 */
public class WsServerInitializer extends ChannelInitializer<SocketChannel> {
    @Override
    protected void initChannel(SocketChannel ch) throws Exception {
        ChannelPipeline pipeline = ch.pipeline();
        // WebSocket 基于 Http 协议, 所以要有 Http 的编解码器
        pipeline.addLast(new HttpServerCodec());
        // 对写大数据流的支持
        pipeline.addLast(new ChunkedWriteHandler());
        // 对 HTTP Message 进行聚合,聚合成 FullHttpRequest 或 FullHttpResponse
        // 几乎在 netty中的编程,都会使用到此 handler
        pipeline.addLast(new HttpObjectAggregator(65536));

        //============== 以上是支持 HTTP 协议 ==============

        /*
         * webSocket 服务器处理的协议,用于指定给客户端连接访问的路由
         * 这个 handler 会帮你处理一些繁重复杂的事情.
         * 会帮你处理握手动作: Shandshaking(close, ping, pong)  ping + pong = 心跳
         * 对于 webSocket 来讲,都是以 frames 进行传输的, 不同的数据类型对应的 frames 也不同
         */
        pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));

        // 自定义的 handler
        pipeline.addLast(new ChatHandler());
    }
}
